
#ifndef __EDNN_LAYERS_H__
#define __EDNN_LAYERS_H__



#include "ednn.h"



struct ednn_shape3d shape3d(ednn_shape_t h, 
    ednn_shape_t w, ednn_shape_t c);
#define kernel(h, w)    shape3d(h, w, 1)
#define stride(h, w)    shape3d(h, w, 1)
#define dilation(h, w)  shape3d(h, w, 1)

#define ednn_ceilif(x,y)    ((x+y-1)/y)

#include "ednn_input.h"
#include "ednn_output.h"
#include "ednn_activation.h"
#include "ednn_dense.h"
#include "ednn_softmax.h"
#include "ednn_conv2d.h"
#include "ednn_maxpool.h"
#include "ednn_flatten.h"



#endif /* __EDNN_LAYERS_H__ */
